package org.skywork.top.core.algorithm.array;

import java.util.Arrays;

/**
 * 加一
 */
public class no_66 {

    /**
     * 给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。
     *
     * 最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。
     *
     * 你可以假设除了整数 0 之外，这个整数不会以零开头。
     */
    public static int[] plusOne(int[] digits) {

        /**
         * 示例 1：
         *
         * 输入：digits = [1,2,3]
         * 输出：[1,2,4]
         * 解释：输入数组表示数字 123。
         *
         * 示例 2：
         *
         * 输入：digits = [4,3,2,1]
         * 输出：[4,3,2,2]
         * 解释：输入数组表示数字 4321。
         *
         * 示例 3：
         *
         * 输入：digits = [9]
         * 输出：[1,0]
         * 解释：输入数组表示数字 9。
         * 加 1 得到了 9 + 1 = 10。
         * 因此，结果应该是 [1,0]。
         */

        if(digits[digits.length - 1] == 9){
            int[] newDigits = new int[digits.length + 1];
            boolean flag = digits[0] == 9;
            boolean add = false;
            for(int i = digits.length - 1; i >= 0; i--){
                if(i == digits.length - 1 && digits[i] == 9){
                    digits[i] = 0;
                    add = true;
                }else{
                    if(digits[i] == 9 && add){
                        digits[i] = 0;
                    }else{
                        if(add){
                            digits[i] = digits[i] + 1;
                            add = false;
                        }else{
                            digits[i] = digits[i];
                        }
                    }
                }
            }
            if(flag && add){
                newDigits[0] = 1;
                System.arraycopy(digits, 0, newDigits, 1, digits.length);
                digits = newDigits;
            }
        }else{
            digits[digits.length - 1] = digits[digits.length - 1] + 1;
            return digits;
        }
        return digits;
    }

    public static void main(String[] args) {
        int[] digits = {9,8,9};
        System.out.println(Arrays.toString(plusOne(digits)));
    }

}
